#!/usr/bin/python
from transformers import pipeline

model_name = "mfeb/albert-xxlarge-v2-squad2"

eqa_model = pipeline('question-answering', model=model_name)
question_context = {}
question_context["context"] = "The Amazon rainforest (Portuguese: " \
          "Floresta Amazônica or Amazônia; Spanish: Selva Amazónica" \
          ", Amazonía or usually Amazonia; French: Forêt amazonienne;" \
          " Dutch: Amazoneregenwoud), also known in English as Amazonia" \
          " or the Amazon Jungle, is a moist broadleaf forest that covers" \
          " most of the Amazon basin of South America. This basin" \
          " encompasses 7,000,000 square kilometres (2,700,000 sq mi)" \
          ", of which 5,500,000 square kilometres (2,100,000 sq mi)" \
          " are covered by the rainforest. This region includes territory" \
          " belonging to nine nations. The majority of the forest is contained" \
          " within Brazil, with 60% of the rainforest, followed by Peru with" \
          " 13%, Colombia with 10%, and with minor amounts in Venezuela, Ecuador" \
          ", Bolivia, Guyana, Suriname and French Guiana. States or departments" \
          " in four nations contain 'Amazonas' in their names. The" \
          " Amazon represents over half of the planet's remaining" \
          " rainforests, and comprises the largest and most biodiverse" \
          " tract of tropical rainforest in the world, with an estimated " \
          "390 billion individual trees divided into 16,000 species."
question_context["question"] = "Which name is also used to describe the Amazon rainforest in English?"
answer_confidence = eqa_model(question_context)['score']
print(answer_confidence)